#ifndef _THREAD_HPP__
#define _THREAD_HPP__

#include <iostream>
#include <string>
#include <pthread.h>
#include <functional>
#include <sys/types.h>
#include <unistd.h>

// version 1
// namespace ThreadModule
// {
//     // 执行任务的方法
//     using func_t = std::function<void()>;
//     static int number = 1;

//     // 线程状态-枚举
//     enum class TSTATUS
//     {
//         NEW,
//         RUNNING,
//         STOP
//     };
    
//     class Thread
//     {
//     private:
//         // 成员方法，具备this指针，置为static之后就不具备this指针了
//         static void* Routine(void* args)
//         {
//             // t就拿到了this指针
//             Thread* t = static_cast<Thread *>(args);
//             t->_status = TSTATUS::RUNNING;
//             t->_func(); // 就可以执行相应的类内方法了 
//             return nullptr;
//         }
//         void EnableDetach() // 线程是否能够分离
//         {
//             _joinable = false;
//         }
//     public:
//         // 线程要执行的方法直接传进来
//         Thread(func_t func)
//             : _func(func)
//             , _status(TSTATUS::NEW)
//             , _joinable(true)
//         {
//             _name = "Thread-" + std::to_string(number);
//             _pid = getpid();
//         }
//         bool Start()
//         {
//             if(_status != TSTATUS::RUNNING)
//             {
//                 int n = ::pthread_create(&_tid, nullptr, Routine, this);  // 将this指针通过参数传过去
//                 if(n != 0)  return false;
//                 return true;
//             }
//             return false;
//         }
//         bool Stop()
//         {
//             if(_status == TSTATUS::RUNNING)
//             {
//                 int n = ::pthread_cancel(_tid);
//                 if(n != 0)  return false;
//                 _status = TSTATUS::STOP;
//                 return true;
//             }
//             return false;
//         }
//         bool Join()
//         {
//             if(_joinable)
//             {
//                 int n = ::pthread_join(_tid, nullptr);
//                 if(n != 0)  return false;
//                 _status = TSTATUS::STOP;
//                 return true;
//             }
//             return false;
//         }
//         void Detach()
//         {
//             EnableDetach();
//             pthread_detach(_tid);
//         }
//         bool IsJoinable()   
//         {
//             return _joinable;
//         }
//         std::string Name() { return _name; }
//         ~Thread() {}

//     private:
//         std::string _name; // 线程名字
//         pthread_t _tid;    // 线程id
//         pid_t _pid;        // 进程id
//         bool _joinable;    // 是否是分离状态，默认不是
//         func_t _func;      // 线程未来要执行的方法
//         TSTATUS _status;   // 线程状态
//     };
// }

// version 2    传参版本
namespace ThreadModule
{
    // 执行任务的方法
    static int number = 1;

    // 线程状态-枚举
    enum class TSTATUS
    {
        NEW,
        RUNNING,
        STOP
    };
    template <typename T>
    class Thread
    {
        using func_t = std::function<void(T)>;
    private:
        // 成员方法，具备this指针，置为static之后就不具备this指针了
        static void* Routine(void* args)
        {
            // t就拿到了this指针
            Thread<T>* t = static_cast<Thread<T> *>(args);
            t->_status = TSTATUS::RUNNING;
            t->_func(t->_data); // 就可以执行相应的类内方法了 
            return nullptr;
        }
        void EnableDetach() // 线程是否能够分离
        {
            _joinable = false;
        }
    public:
        // 线程要执行的方法直接传进来
        Thread(func_t func, T data)
            : _func(func)
            , _data(data)
            , _status(TSTATUS::NEW)
            , _joinable(true)
        {
            _name = "Thread-" + std::to_string(number);
            _pid = getpid();
        }
        bool Start()
        {
            if(_status != TSTATUS::RUNNING)
            {
                int n = ::pthread_create(&_tid, nullptr, Routine, this);  // 将this指针通过参数传过去
                if(n != 0)  return false;
                return true;
            }
            return false;
        }
        bool Stop()
        {
            if(_status == TSTATUS::RUNNING)
            {
                int n = ::pthread_cancel(_tid);
                if(n != 0)  return false;
                _status = TSTATUS::STOP;
                return true;
            }
            return false;
        }
        bool Join()
        {
            if(_joinable)
            {
                int n = ::pthread_join(_tid, nullptr);
                if(n != 0)  return false;
                _status = TSTATUS::STOP;
                return true;
            }
            return false;
        }
        void Detach()
        {
            EnableDetach();
            pthread_detach(_tid);
        }
        bool IsJoinable()   
        {
            return _joinable;
        }
        std::string Name() { return _name; }
        ~Thread() {}

    private:
        std::string _name; // 线程名字
        pthread_t _tid;    // 线程id
        pid_t _pid;        // 进程id
        bool _joinable;    // 是否是分离状态，默认不是
        func_t _func;      // 线程未来要执行的方法
        TSTATUS _status;   // 线程状态
        T _data;
    };
}
#endif